../../../../../../../../models/demos/ttnn_falcon7b/tests/test_falcon_rotary_embedding.py